package org.lmj.client;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import org.lmj.bean.RpcRequest;
import org.lmj.bean.RpcResponse;
import org.lmj.protocol.RpcDecoder;
import org.lmj.protocol.RpcEncoder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class RpcClient extends SimpleChannelInboundHandler<RpcResponse> {

    private static ThreadPoolExecutor threadPoolExecutor = new ThreadPoolExecutor(16, 16,
            600L, TimeUnit.SECONDS, new ArrayBlockingQueue<Runnable>(65536));

    private static final Logger LOGGER = LoggerFactory.getLogger(RpcClient.class);

    private String host;
    private int port;

    private RpcResponse response;

    private final Object obj = new Object();

    public RpcClient(String host, int port) {
        this.host = host;
        this.port = port;
    }

    public static void submit(Runnable task) {
        threadPoolExecutor.submit(task);
    }

    @Override
    public void channelRead0(ChannelHandlerContext ctx, RpcResponse response) throws Exception {
        /**
         * netty进行网络通信，在channelRead0去接收读入/写出的结果。
         * netty会自动的将字节按照协议编码/解码的要求将其封装到RpcResponse/RpcRequest
         */
        this.response = response;

        synchronized (obj) {
            obj.notifyAll(); // 收到响应，唤醒线程
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        LOGGER.error("client caught exception", cause);
        ctx.close();
    }

    public RpcResponse send(RpcRequest request) throws Exception {
        /**
         * 处理网络数据传输通道里的信息（数据编码/解码）
         * 然后在继承了SimpleChannelInboundHandler的
         * 子类的channelRead0方法里可以处理/接受上述流程处理后的结果（即request/response）
         */
        EventLoopGroup group = new NioEventLoopGroup();
        try {
            Bootstrap bootstrap = new Bootstrap();
            bootstrap.group(group).channel(NioSocketChannel.class)
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        public void initChannel(SocketChannel channel) throws Exception {
                            channel.pipeline()
                                    .addLast(new RpcEncoder(RpcRequest.class)) // 将 RPC 请求进行编码（为了发送请求）
                                    .addLast(new RpcDecoder(RpcResponse.class)) // 将 RPC 响应进行解码（为了处理响应）
                                    .addLast(RpcClient.this); // 使用 RpcClient 发送 RPC 请求
                        }
                    })
                    .option(ChannelOption.SO_KEEPALIVE, true);

            ChannelFuture future = bootstrap.connect(host, port).sync();
            /**
             * //将request对象写入outbundle处理后发出（即RpcEncoder编码器）
             *             future.channel().writeAndFlush(request).sync();
             *
             *             // 用线程等待的方式决定是否关闭连接
             *             // 其意义是：先在此阻塞，等待获取到服务端的返回后，被唤醒，从而关闭网络连接
             */
            future.channel().writeAndFlush(request).sync();

            /**
             * 这里每次使用代理远程调用服务，从Zookeeper上获取可用的服务地址，通过RpcClient send一个Request，
             * 等待该Request的Response返回。这里原文有个比较严重的bug，在原文给出的简单的Test中是很难测出来的，
             * 原文使用了obj的wait和notifyAll来等待Response返回，会出现“假死等待”的情况：一个Request发送出去后，
             * 在obj.wait()调用之前可能Response就返回了，这时候在channelRead0里已经拿到了Response并且
             * obj.notifyAll()已经在obj.wait()之前调用了，这时候send后再obj.wait()就出现了假死等待，
             * 客户端就一直等待在这里。使用CountDownLatch可以解决这个问题。
             *
             * 注意：这里每次调用的send时候才去和服务端建立连接，使用的是短连接，这种短连接在高并发时会有连接数问题，也会影响性能。
             */

            /**
             * // 用线程等待的方式决定是否关闭连接
             *             // 其意义是：先在此阻塞，等待获取到服务端的返回后，被唤醒，从而关闭网络连接
             *             这是一个短连接
             */
            synchronized (obj) {
                obj.wait(); // 未收到响应，使线程等待
            }

            /**
             *  private final Object obj = new Object();
             *      在上文里面有代码
             *          obj.notifyAll(); // 收到响应，唤醒线程
             *          }
             */

            if (response != null) {
                //服务器同步连接断开时,这句代码才会往下执行
                future.channel().closeFuture().sync();
            }
            return response;
        } finally {
            group.shutdownGracefully();
        }
    }
}
